package com.chif.hdfs;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.*;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Arrays;

/**
 * 客户端代码常用套路
 * 1.获取一个客户端对象
 * 2.执行相关的操作命令
 * 3.关闭资源
 * HDFS Zookeeper
 */
public class HdfsClient {

    private FileSystem fs;

    @Before
    public void init() throws URISyntaxException, IOException, InterruptedException {
        //连接的集群NN地址
        URI uri = new URI("hdfs://hadoop102:8020");
        //配置文件
        Configuration configuration = new Configuration();
        configuration.set("dfs.replication","2");//副本数量
        //用户
        String user="liudachu";

        //1.获取到了客户端对象
        fs = FileSystem.get(uri,configuration,user);
    }

    @After
    public void close() throws IOException {
        //3.关闭资源
        fs.close();
    }




    //创建目录
    @Test
    public void testmkdir() throws URISyntaxException, IOException, InterruptedException {
        //2.创建一个文件夹
        fs.mkdirs(new Path("/xiyou/huaguoshan2"));
    }



    //上传文件
    @Test
    public void testPut() throws IOException {
        //参数解读  delSrc：表示删除源数据 overwriter：是否允许覆盖 src：元数据路径 dst：目的地路径
        fs.copyFromLocalFile(false,true,new Path("D:\\sunwukong.txt"),new Path("hdfs://hadoop102/xiyou/huaguoshan"));
    }

    /**
     * 参数优先级 低=>高
     * hdfs-default.xml => hdfs-site.xml => 在项目资源目录下的配置文件 => 代码里面的configuration配置
      */

    //下载文件
    @Test
    public void testGet() throws IOException {
        //参数解读  delSrc：是否删除源数据 src：元数据路径 dst：目的地路径  useRawLocalFileSystem:是否开启校验 true=>不开启
        fs.copyToLocalFile(false,new Path("hdfs://hadoop102/xiyou/huaguoshan"),new Path("D:\\"),true);
    }


    //删除
    @Test
    public void testRm() throws IOException {
        //参数解读  Path：要删除的路径 recursive:是否递归删除
        fs.delete(new Path("/xiyou/huaguoshan/sunwukong.txt"),false);

        //删除空目录
        fs.delete(new Path("/xiyou/huaguoshan"),false);


        //删除非空目录
        fs.delete(new Path("/xiyou"),true);

    }

    //文件的更名和移动
    @Test
    public void testmv() throws IOException {

        //对文件名称的修改
        //参数解读 src：源文件路径 dst：目标路径
        //fs.rename(new Path("/input/word.txt"),new Path("/input/liudachu.txt"));

        //文件的移动和更名
        //参数解读 src：源文件路径 dst：目标路径
        //fs.rename(new Path("/input/liudachu.txt"),new Path("/avi.txt"));


        //目录更名
        fs.rename(new Path("/input"),new Path("/input2"));

    }



    //获取文件详细信息(遍历文件列表)
    @Test
    public void fileDetail() throws IOException {

        //获取所有文件信息 路径+是否递归
        RemoteIterator<LocatedFileStatus> listFiles = fs.listFiles(new Path("/"), true);

        while (listFiles.hasNext()){
            LocatedFileStatus fileStatus = listFiles.next();
            System.out.println("========"+fileStatus.getPath()+"==========");
            System.out.println(fileStatus.getPermission());
            System.out.println(fileStatus.getOwner());
            System.out.println(fileStatus.getGroup());
            System.out.println(fileStatus.getLen());
            System.out.println(fileStatus.getModificationTime());
            System.out.println(fileStatus.getReplication());
            System.out.println(fileStatus.getBlockSize());
            System.out.println(fileStatus.getPath().getName());


            //获取块信息
            BlockLocation[] blockLocations = fileStatus.getBlockLocations();
            System.out.println(Arrays.toString(blockLocations));
        }

    }



    //判断是文件夹还是文件
    @Test
    public void testFile() throws IOException {
        FileStatus[] fileStatuses = fs.listStatus(new Path("/"));

        for (FileStatus fileStatus : fileStatuses) {
            if (fileStatus.isFile()) {
                System.out.println("文件："+fileStatus.getPath().getName());
            }  else {
                System.out.println("目录："+fileStatus.getPath().getName());
            }
        }
    }

}
